# 处理图片的脚本
# 提取文件夹下所有图片的名字，同时将所有图片改成jpg格式

import os

# 指定图片文件夹路径
image_folder_path = os.path.abspath(os.path.join(os.getcwd(), "pydataset_plus"))
assert os.path.exists(image_folder_path), "{} path does not exist.".format(image_folder_path)  # 断言警告

# 获取图片文件夹下的所有文件
file_list = os.listdir(image_folder_path)

# 初始化计数器
counter = 100000

# 初始化image_list
image_list = []

for file in file_list:
    # 构造新的文件名
    new_name = str(counter) + '.jpg'

    # 重命名文件
    os.rename(os.path.join(image_folder_path, file), os.path.join(image_folder_path, new_name))

    # 将新的文件名添加到列表
    image_list.append(new_name)

    # 更新计数器
    counter += 1

# 将图片名称写入 txt 文件
with open('img.txt', 'w', encoding='utf-8') as output_file:
    for image_name in image_list:
        output_file.write(f'{image_name}\n')